package com.hmsm.server.util;

import java.io.File;
import java.io.FileInputStream;
import java.io.IOException;
import java.io.InputStream;
import java.io.UnsupportedEncodingException;
import java.security.MessageDigest;
import java.security.NoSuchAlgorithmException;

import org.apache.commons.logging.Log;
import org.apache.commons.logging.LogFactory;

/**
 * @描述:MD5生成器
 * @作者:Lzyyj
 * @author Lzyyj
 * @创建日期:2014-4-1 下午03:37:32
 */
public class MD5 {
	
	private static Log log = LogFactory.getLog(MD5.class);
	
	// 用来将字节转换成 16 进制表示的字符
	private static char hexDigits[] = { '0', '1', '2', '3', '4', '5', '6', '7', '8', '9', 'a', 'b', 'c', 'd', 'e', 'f' };

	/*
	 * 把byte[]数组转换成十六进制字符串表示形式
	 * 
	 * @param tmp 要转换的byte[]
	 * 
	 * @return 十六进制字符串表示形式
	 */
	private static String byteToHexString(byte[] tmp) {
		String s;
		// 用字节表示就是 16 个字节
		char str[] = new char[16 * 2]; // 每个字节用 16 进制表示的话，使用两个字符，
		// 所以表示成 16 进制需要 32 个字符
		int k = 0; // 表示转换结果中对应的字符位置
		for (int i = 0; i < 16; i++) { // 从第一个字节开始，对 MD5 的每一个字节
			// 转换成 16 进制字符的转换
			byte byte0 = tmp[i]; // 取第 i 个字节
			str[k++] = hexDigits[byte0 >>> 4 & 0xf]; // 取字节中高 4 位的数字转换,
			// >>> 为逻辑右移，将符号位一起右移
			str[k++] = hexDigits[byte0 & 0xf]; // 取字节中低 4 位的数字转换
		}
		s = new String(str); // 换后的结果转换为字符串
		return s;
	}

	/*
	 * 对message进行MD5加密
	 * @param message	要加密的信息，使用系统默认编码格式进行编码加密
	 * @return
	 * @throws NoSuchAlgorithmException
	 */
	public static String getMD5(String message){
		return getMD5(message, null);
	}
	
	/**
	 * 对message进行MD5加密
	 * @param message	要加密的信息
	 * @param charsetname	要使用的编码格式, 如果编码失败，会改用系统默认编码格式进行编码
	 * @return
	 * @throws NoSuchAlgorithmException
	 */
	public static String getMD5(String message, String charsetname){
		MessageDigest md;
		try {
			md = MessageDigest.getInstance("MD5");
		} catch (NoSuchAlgorithmException e1) {
			e1.printStackTrace();
			log.error("MD5加密失败，请检查环境配置！");
			return null;
		}
		
		byte[] b;
		try {
			b = md.digest(charsetname==null||charsetname.trim().length()==0?message.getBytes():message.getBytes(charsetname));
		} catch (UnsupportedEncodingException e) {
			b = md.digest(message.getBytes());
			log.warn("MD5 加密前字符串进行 "+charsetname+" 编码失败，改为默认编码格式进行！");
		}
		return byteToHexString(b);
	}
	
	/**
	 * 获取文件的 MD5 编码，支持大文件
	 * @param file
	 * @return
	 * @throws IOException
	 * @throws NoSuchAlgorithmException
	 */
    public static String getMD5(File file) throws IOException, NoSuchAlgorithmException{
        
        InputStream ins = new FileInputStream(file);
        
        byte[] buffer = new byte[8192];
        MessageDigest md5 = MessageDigest.getInstance("MD5");
        
        int len;
        while((len = ins.read(buffer)) != -1){
            md5.update(buffer, 0, len);
        }

        ins.close();
//        也可以用apache自带的计算MD5方法
//        return DigestUtils.md5Hex(md5.digest());
        
//        自己写的转计算MD5方法
        return byteToHexString(md5.digest());
    }
    
	
	/**
	 * 判断是否是一个MD5加密字符串
	 * @param message
	 * @return
	 */
	public static boolean isMD5String(String message){
		boolean result = false;
		
		if (message!=null){
			String testStr = message.trim().toLowerCase();
			if (testStr.length()==16 || testStr.length()==32){
				result = true;
				for(int i=0;i<testStr.length();i++){
					if (!(testStr.charAt(i)>='0'&&testStr.charAt(i)<='9'||testStr.charAt(i)>='a'&&testStr.charAt(i)<='h')){
						result = false;
						break;
					}
				}
			}
		}
		
		return result;
	}

	public static String getMD5(byte[] bytes) {
		MessageDigest md;
		try {
			md = MessageDigest.getInstance("MD5");
			byte[] result = md.digest(bytes);
			return byteToHexString(result);
		} catch (NoSuchAlgorithmException e) {
			e.printStackTrace();
		}

		return null;
	}

}
